package com.kc.common.utils;

import com.kc.common.exception.CustomException;
import lombok.extern.slf4j.Slf4j;

import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.util.Map;

/**
 * 签名工具类
 *
 * @author TP
 * @date 2021-02-06 09:04
 */
@Slf4j
public class SignUtils {

    /**
     * 生成签名
     *
     * @param url    请求url，例如：/user/add
     * @param params 请求参数map
     * @param key    加密密钥
     * @return 签名串
     */
    public static String generateSign(String url, Map<String, Object> params, String key) {
        if (StringUtils.isBlank(key)) {
            throw new CustomException("请指定加密密钥");
        }
        if (params == null) {
            throw new CustomException("请求参数不能为空，请至少提供一个kcTimeStamp参数");
        }
        params.put("kc_interface_security_key", key);
        String linkString = EncryptUtils.createLinkString(params, false);
        params.remove("kc_interface_security_key");
        return generateMd5(url + linkString);
    }

    /**
     * 获取MD5指纹
     *
     * @param string 要加密的数据
     * @return 指纹
     */
    private static String generateMd5(String string) {
        try {
            char[] hexDigits = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
            byte[] inputBytes = string.getBytes(StandardCharsets.UTF_8);
            MessageDigest mdInst = MessageDigest.getInstance("md5");
            // 使用指定的字节更新摘要
            mdInst.update(inputBytes);
            // 获得密文
            byte[] md = mdInst.digest();
            // 把密文转换成十六进制的字符串形式
            int j = md.length;
            char[] str = new char[j * 2];
            int k = 0;
            for (int i = 0; i < j; i++) {
                byte byte0 = md[i];
                // 将每个数(int)b进行双字节加密
                str[k++] = hexDigits[byte0 >>> 4 & 0xf];
                str[k++] = hexDigits[byte0 & 0xf];
            }
            return new String(str);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }
}
